package cContext

import (
	"github.com/gin-gonic/gin"
	"google.golang.org/grpc/metadata"

	"gitee.com/csingo/cCommon"
)

func SetMetadata(ctx *gin.Context, k, v string) {
	md, ok := ctx.Get(cCommon.GRPC_Metadata)
	if !ok {
		md = metadata.MD{}
	}

	md.(metadata.MD).Set(k, v)
}

func GetMetadata(ctx *gin.Context, k string) string {
	md, ok := ctx.Get(cCommon.GRPC_Metadata)
	if !ok {
		return ""
	}

	values := md.(metadata.MD).Get(k)
	if len(values) > 0 {
		return values[0]
	}

	return ""
}
